import pymongo
import redis
redis_client = redis.Redis()
mongo_handler = pymongo.MongoClient().chapter_8.news


def init():
    all_title = mongo_handler.distinct('title') # 获取所有新闻标题。
    redis_client.sadd('news_title', *all_title) # 把新闻标题全部添加到 Redis 中名为news_title 的集合中。


def need_insert_news(news_title):
    # 添加并判断新闻标题是否己经在news_title集合中。如果己经存在,则返回0;如果不存在,则返回1,并将其添加进入Redis集合中。
    if redis_client.sadd('news_title', news_title) == 1:
        return True
    return False
'''
由于Redis的读写速度远远快于MongoDB,因此使用Redis可避免频繁读取MongoDB从而大大提高程序性能。
'''